package com.ly.web.controller.system;

import java.util.Iterator;
import java.util.List;

import com.ly.common.PageParam;
import com.ly.common.annotation.Log;
import com.ly.common.constant.UserConstants;
import com.ly.common.core.controller.BaseController;
import com.ly.common.core.domain.AjaxResult;
import com.ly.common.core.domain.entity.SysDept;
import com.ly.common.enums.BusinessType;
import com.ly.common.utils.SecurityUtils;
import com.ly.common.utils.StringUtils;
import com.ly.common.utils.erp.SnowflakeIdWorker;
import com.ly.system.service.ISysDeptService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ArrayUtils;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;

/**
 * 部门信息
 * @author ruoyi
 */
@Api (value = "Dept", tags = "部门管理")
@Slf4j
@AllArgsConstructor
@RestController
@RequestMapping ("/system/dept")
public class SysDeptController extends BaseController {

    @Resource
    private ISysDeptService deptService;

    /**
     * 获取部门列表
     */
    @PreAuthorize ("@ss.hasPermi('system:dept:list')")
    @GetMapping ("/list")
    @ApiOperation ("获取部门列表")
    public AjaxResult list (SysDept dept) {
        List<SysDept> depts = deptService.selectDeptList ( dept );
        PageParam pageParam = new PageParam ();
        pageParam.setRecords ( depts ).setTotal ( deptService.queryCount ( dept ) );
        return AjaxResult.success ( pageParam );
    }

    /**
     * 查询部门列表（排除节点）
     */
    @PreAuthorize ("@ss.hasPermi('system:dept:list')")
    @GetMapping ("/list/exclude/{deptId}")
    @ApiOperation ("查询部门列表（排除节点）")
    public AjaxResult excludeChild (@PathVariable (value = "deptId", required = false) String deptId) {
        List<SysDept> depts = deptService.selectDeptList ( new SysDept () );
        Iterator<SysDept> it = depts.iterator ();
        while ( it.hasNext () ) {
            SysDept d = (SysDept) it.next ();
            if ( d.getDeptId ().equals ( deptId ) || ArrayUtils.contains ( StringUtils.split ( d.getAncestors (), "," ), deptId ) ) {
                it.remove ();
            }
        }
        return AjaxResult.success ( depts );
    }

    /**
     * 根据部门编号获取详细信息
     */
    @PreAuthorize ("@ss.hasPermi('system:dept:query')")
    @GetMapping (value = "/{deptId}")
    @ApiOperation ("根据部门编号获取详细信息")
    public AjaxResult getInfo (@PathVariable String deptId) {
        return AjaxResult.success ( deptService.selectDeptById ( deptId ) );
    }

    /**
     * 获取部门下拉树列表
     */
    @GetMapping ("/treeselect")
    @ApiOperation ("获取部门下拉树列表")
    public AjaxResult treeselect (SysDept dept) {
        List<SysDept> depts = deptService.selectDeptList ( dept );
        return AjaxResult.success ( deptService.buildDeptTreeSelect ( depts ) );
    }

    /**
     * 加载对应角色部门列表树
     */
    @GetMapping (value = "/roleDeptTreeselect/{roleId}")
    @ApiOperation ("加载对应角色部门列表树")
    public AjaxResult roleDeptTreeselect (@PathVariable ("roleId") String roleId) {
        List<SysDept> depts = deptService.selectDeptList ( new SysDept () );
        AjaxResult ajax = AjaxResult.success ();
        ajax.put ( "checkedKeys", deptService.selectDeptListByRoleId ( roleId ) );
        ajax.put ( "depts", deptService.buildDeptTreeSelect ( depts ) );
        return ajax;
    }

    /**
     * 新增部门
     */
    @PreAuthorize ("@ss.hasPermi('system:dept:add')")
    @Log (title = "部门管理", businessType = BusinessType.INSERT)
    @PostMapping
    @ApiOperation ("新增部门")
    public AjaxResult add (@Validated @RequestBody SysDept dept) {
        if ( UserConstants.NOT_UNIQUE.equals ( deptService.checkDeptNameUnique ( dept ) ) ) {
            return AjaxResult.error ( "新增部门'" + dept.getDeptName () + "'失败，部门名称已存在" );
        }
        dept.setDeptId ( SnowflakeIdWorker.getInstance ().nextStringId () );
        return toAjax ( deptService.insertDept ( dept ) );
    }

    /**
     * 修改部门
     */
    @PreAuthorize ("@ss.hasPermi('system:dept:edit')")
    @Log (title = "部门管理", businessType = BusinessType.UPDATE)
    @PutMapping
    @ApiOperation ("修改部门")
    public AjaxResult edit (@Validated @RequestBody SysDept dept) {
        if ( UserConstants.NOT_UNIQUE.equals ( deptService.checkDeptNameUnique ( dept ) ) ) {
            return AjaxResult.error ( "修改部门'" + dept.getDeptName () + "'失败，部门名称已存在" );
        } else if ( dept.getParentId ().equals ( dept.getDeptId () ) ) {
            return AjaxResult.error ( "修改部门'" + dept.getDeptName () + "'失败，上级部门不能是自己" );
        } else if ( StringUtils.equals ( UserConstants.DEPT_DISABLE, dept.getStatus () )
                && deptService.selectNormalChildrenDeptById ( dept.getDeptId () ) > 0 ) {
            return AjaxResult.error ( "该部门包含未停用的子部门！" );
        }
        dept.setUpdateBy ( SecurityUtils.getUsername () );
        return deptService.updateDept ( dept );
    }

    /**
     * 删除部门
     */
    @PreAuthorize ("@ss.hasPermi('system:dept:remove')")
    @Log (title = "部门管理", businessType = BusinessType.DELETE)
    @DeleteMapping ("/{deptId}")
    @ApiOperation ("删除部门")
    public AjaxResult remove (@PathVariable String deptId) {
        if ( deptService.hasChildByDeptId ( deptId ) ) {
            return AjaxResult.error ( "存在下级部门,不允许删除" );
        }
        if ( deptService.checkDeptExistUser ( deptId ) ) {
            return AjaxResult.error ( "部门存在用户,不允许删除" );
        }
        return toAjax ( deptService.deleteDeptById ( deptId ) );
    }
}
